package com.atguigu.study.config;

import dev.langchain4j.community.model.dashscope.WanxImageModel;
import dev.langchain4j.model.chat.ChatModel;
import dev.langchain4j.model.openai.OpenAiChatModel;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * @author Marco
 * @Description
 * @date 2025/8/18 23:16
 * @email 3293336923@qq.com
 */
@Configuration
public class LLMConfig {


    @Bean
    public ChatModel    chatModel(){

       return   OpenAiChatModel.builder()
                .apiKey(System.getenv("ALIQW"))
//                qwen-vl-max 是一个多模态大模型，支持文本和图片 的结合输入，适用于视觉-语言任务。
                .modelName("qwen-vl-max")
                .baseUrl("https://dashscope.aliyuncs.com/compatible-mode/v1")
                .build();
    }


    /***
     *  使用 阿里万象大模型
     * @return
     */
    @Bean
    public WanxImageModel    wanxImageModel(){
        return
                WanxImageModel.builder()
                        .apiKey(System.getenv("ALIQW"))
                        .modelName("wanx2.1-t2i-turbo")  // 图片生成 https://help.aliyun.com/zh/model-studio/text-to-image
                        .build();
    }
}
